package com.it.data_tree.tree.sort;

/**
 * @Author: CaoYouGen
 * @DateTime: 2023/05/04/10:12
 * @注释: 归并排序
 **/
public class MergeSort {
    public static void mergeSort(int[] arr) {
        // 判断数组是否为空或长度小于等于1，若是则直接返回
        if (arr == null || arr.length <= 1) {
            return;
        }
        // 对整个数组进行归并排序
        mergeSort(arr, 0, arr.length - 1);
    }

    /*
        [2, 1, 6]

        [2, 1]  [6]

        [2] [1]
    * */
    private static void mergeSort(int[] arr, int left, int right) {
        // 如果左边索引大于等于右边索引，说明数组只有一个元素或没有元素，返回即可
        if (left >= right) {
            return;
        }
        // 计算数组的中间位置
        int mid = left + ((right - left) >>> 1);
        // 对左半部分进行归并排序
        mergeSort(arr, left, mid);
        // 对右半部分进行归并排序
        mergeSort(arr, mid + 1, right);
        // 将左右两部分归并
        merge(arr, left, mid, right);
    }

    private static void merge(int[] arr, int left, int mid, int right) {
        // 临时数组，用于存储归并结果
        int[] tmp = new int[right - left + 1];
        int i = left, j = mid + 1, k = 0;
        // 依次比较左半部分和右半部分的元素大小，将较小的元素存入临时数组中
        while (i <= mid && j <= right) {
            if (arr[i] <= arr[j]) {
                tmp[k++] = arr[i++];
            } else {
                tmp[k++] = arr[j++];
            }
        }
        // 将左半部分剩余的元素存入临时数组中
        while (i <= mid) {
            tmp[k++] = arr[i++];
        }
        // 将右半部分剩余的元素存入临时数组中
        while (j <= right) {
            tmp[k++] = arr[j++];
        }
        // 将临时数组中的元素复制回原数组中
        System.arraycopy(tmp, 0, arr, left, tmp.length);
    }
}
